#!/bin/bash

# 进入文件目录
resnet=dirname
cd ~/$dirname

# 激活虚拟环境
source resnet/bin/activate

#如果脚本不存在就下载
if [ ! -f train_resnet50.py ]; then
    wget https://gitee.com/huangsheng32/amdlinux/raw/master/amdgpu%20resnet50/train_resnet50.py
fi

#
python train_resnet50.py --batch_size=16 --epochs=10